package mine.code.question.动态规划;

import org.junit.Test;

/**
 * 给定一个整数数组 nums ，你可以对它进行一些操作。
 * <p>
 * 每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除每个等于 nums[i] - 1 或 nums[i] + 1 的元素。
 * <p>
 * 开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
 * <p>
 * 示例:
 * 输入: nums = [3, 4, 2]
 * 输出: 6
 * 解释:
 * 删除 4 来获得 4 个点数，因此 3 也被删除。
 * 之后，删除 2 来获得 2 个点数。总共获得 6 个点数。
 *
 * @author caijinnan
 * @date 2020/5/26 11:41
 */
public class 删除与获得点数 {


    @Test
    public void run() {
//        int[] nums = {3, 4, 2};
//        int[] nums = {2, 2, 3, 3, 3, 4};
//        int[] nums = {3,1};
        int[] nums = {1};
        System.out.println(deleteAndEarn(nums));
    }


    //数组转换->打家劫舍->不能连续取相领数值
    public int deleteAndEarn(int[] nums) {
        int[] newArr = new int[10001];
        for (int num : nums) {
            newArr[num]++;
        }
        //打家劫舍dp
        //dp[i] = max(dp[i-1],arr[i]+dp[i-2])
        int[] dp = new int[10001];
        dp[0] = newArr[0];
        dp[1] = Math.max(newArr[0], newArr[1]);
        for (int i = 2; i < newArr.length; i++) {
            dp[i] = Math.max(dp[i - 1], dp[i - 2] + i * newArr[i]);
        }
        return dp[dp.length - 1];
    }
}